package backtrack;

import java.util.ArrayList;
import java.util.List;

/**
 * @author kaho
 * @since 2021/1/4
 */
public class _51_N皇后 {

    static int[] queues;
    static List<List<String>> result;

    /**
     *
     * 给定一个整数 n，返回所有不同的 n 皇后问题的解决方案。
     *
     * 每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/n-queens
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     * @param n
     * @return
     */
    public static List<List<String>> solveNQueens(int n) {
        result = new ArrayList<>();
        if(n<1){
            return result;
        }
        queues = new int[n];
        place(0);
        return result;
    }

    private static void place(int row) {
        if (row == queues.length){
            addResult();
            return;
        }
        for (int col = 0; col < queues.length; col++) {
            if(isValid(row, col)){
                queues[row] = col;
                place(row + 1);
            }
        }
    }

    private static boolean isValid(int row, int col) {
        for (int i = 0; i < row; i++) {
            if (queues[i] == col) {
                return false;
            }
            if (row-i == Math.abs(col - queues[i])) {
                return false;
            }
        }
        return true;
    }

    private static void addResult() {
        List<String> r = new ArrayList<>();
        for (int row = 0; row < queues.length; row++) {
            StringBuilder sb = new StringBuilder();
            for (int column = 0; column < queues.length; column++) {
                if (queues[row] == column){
                    sb.append("Q");
                }else {
                    sb.append(".");
                }
            }
            r.add(sb.toString());
        }
        result.add(r);
    }

    public static void main(String[] args) {
        List<List<String>> lists = solveNQueens(4);
    }
}
